import unittest

from Finance_fun.base.get_driver import GetDriver
from Finance_fun.page.page04_my_red_envlope import PageMyRedEnvelope


class MyRedEnvelopeTest(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = GetDriver.get_driver()
        cls.my_red_envelope = PageMyRedEnvelope(cls.driver)
        cls.my_red_envelope.page_click_login_link()
        cls.my_red_envelope.page_login("13129628532","W123456")

    @classmethod
    def tearDownClass(cls):
        GetDriver.quit_driver()

    def test01_my_red_envelope_01(self):
        # 点击交易记录
        self.my_red_envelope.my_red_envelope()
        # 点击获得时间
        self.my_red_envelope.get_time()

    def test02_my_red_envelope_02(self):
        # 点击交易记录
        self.my_red_envelope.my_red_envelope()
        # 点击已冻结
        self.my_red_envelope.ydj_link()
        # 点击过期时间
        self.my_red_envelope.gq_time()

    def test03_my_red_envelope_03(self):
        # 点击交易记录
        self.my_red_envelope.my_red_envelope()
        # 点击已使用
        self.my_red_envelope.ysy_link()
        # 点击过期时间
        self.my_red_envelope.money_link()

    def test04_my_red_envelope_04(self):
        # 点击交易记录
        self.my_red_envelope.my_red_envelope()
        # 点击已使用
        self.my_red_envelope.ygq_link()
        # 点击过期时间
        self.my_red_envelope.money_link()

